package cn.yuitstyle.excel;

import cn.yuitstyle.excel.exception.ExcelException;
import com.alibaba.excel.EasyExcel;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.List;

@Slf4j
public class ExcelDemo {


    public List<ExcelDTO> importExcel(MultipartFile file) {
        ModelDataListener<ExcelDTO> listener = new ModelDataListener<>();
        try {
            EasyExcel.read(file.getInputStream(), ExcelDTO.class, listener)
                    .sheet()
                    // 读取行数
                    .headRowNumber(2)
                    .doRead();
        } catch (IOException e) {
            log.error("", e);
            throw new ExcelException(e.getMessage());
        }
        return listener.getDataList();
    }
}
